package com.powernode.service.impl;

import cn.hutool.core.collection.CollectionUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.powernode.domain.SysUser;
import com.powernode.domain.SysUserRole;
import com.powernode.mapper.SysUserMapper;
import com.powernode.mapper.SysUserRoleMapper;
import com.powernode.service.SysUserRoleService;
import com.powernode.service.SysUserService;
import com.powernode.utils.AuthUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.StringUtils;

import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.stream.Collectors;

@Service
public class SysUserServiceImpl extends ServiceImpl<SysUserMapper, SysUser> implements SysUserService {
    @Autowired
    private SysUserMapper sysUserMapper;
    //
    @Autowired
    private BCryptPasswordEncoder passwordEncoder;

    @Autowired
    private SysUserRoleService sysUserRoleService;

    @Autowired
    private SysUserRoleMapper sysUserRoleMapper;

    /**
     * 新增管理员
     * 1.新增管理员
     * 2.新增管理员与角色的关系
     *
     * @param sysUser
     * @return
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public Integer saveSysUser(SysUser sysUser) {
        // 设置创建用户的ID
        sysUser.setCreateUserId(AuthUtils.getLoginUserId());
        // 设置创建时间
        sysUser.setCreateTime(new Date());
        // 设置店铺ID
        sysUser.setShopId(1L);
        //密码加密
        sysUser.setPassword(passwordEncoder.encode(sysUser.getPassword()));
        // 插入系统用户信息
        int insert = sysUserMapper.insert(sysUser);
        if (insert > 0) {
            // 获取新增用户的ID
            Long userId = sysUser.getUserId();
            // 获取角色ID列表
            List<Long> roleIdList = sysUser.getRoleIdList();
            // 如果角色ID列表不为空且大小不为0
            if (CollectionUtil.isNotEmpty(roleIdList) && roleIdList.size() != 0) {
                // 初始化用户角色关系列表
                List<SysUserRole> sysUserRoleList = new ArrayList<>();
                // 遍历角色ID列表，为每个角色ID创建用户角色关系并插入数据库
                roleIdList.forEach(roleId -> {
                    // 创建一个新的用户角色关系对象
                    SysUserRole sysUserRole = new SysUserRole();
                    // 设置用户ID
                    sysUserRole.setUserId(userId);
                    // 设置角色ID
                    sysUserRole.setRoleId(roleId);
                    // 注意：不建议在循环中操作数据库，大家尽量需要避免在循环中操作数据库
                    // sysUserRoleMapper.insert(sysUserRole);

                    // 将创建的用户角色关系添加到列表中
                    sysUserRoleList.add(sysUserRole);
                });
                // 批量保存用户角色关系
                sysUserRoleService.saveBatch(sysUserRoleList);
            }
        }
        // 返回insert表示操作完成
        return insert;
    }

    /**
     * 根据用户ID查询系统用户信息
     *
     * @param id 用户ID，用于唯一标识一个用户
     * @return 返回包含用户角色信息的SysUser对象如果用户ID对应的角色列表为空，则不设置roleIdList
     */
    @Override
    public SysUser querySysUserInfoByUserId(long id) {
        // 根据用户ID从数据库中查询用户信息
        SysUser sysUser = sysUserMapper.selectById(id);

        // 查询与用户ID关联的所有用户角色记录列表
        List<SysUserRole> sysUserRoleList = sysUserRoleMapper.selectList(new LambdaQueryWrapper<SysUserRole>().eq(SysUserRole::getUserId, id));

        // 检查用户角色列表是否不为空且大小不为0，以避免冗余判断
        if (CollectionUtil.isNotEmpty(sysUserRoleList) && sysUserRoleList.size() != 0) {
            // 从用户角色列表中提取角色ID列表
            List<Long> roleIdList = sysUserRoleList.stream().map(SysUserRole::getRoleId).collect(Collectors.toList());
            // 将角色ID列表设置到用户对象中
            sysUser.setRoleIdList(roleIdList);
        }

        // 返回查询到的用户对象，可能包含角色信息
        return sysUser;
    }

    /**
     * 修改管理员信息
     * 1.删除原有的管理员与角色关系记录
     * 2.添加新的管理员与角色的关系
     * 3.修改管理员信息
     *
     * @param sysUser
     * @return
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public Integer modifySysUserInfo(SysUser sysUser) {
        //获取管理员标识
        Long userId = sysUser.getUserId();
        //删除原有的管理员与角色的关系记录
        sysUserRoleMapper.delete(new LambdaQueryWrapper<SysUserRole>().eq(SysUserRole::getUserId, userId));
        // 获取角色ID列表
        List<Long> roleIdList = sysUser.getRoleIdList();
        // 如果角色ID列表不为空且大小不为0
        if (CollectionUtil.isNotEmpty(roleIdList) && roleIdList.size() != 0) {
            // 初始化用户角色关系列表
            List<SysUserRole> sysUserRoleList = new ArrayList<>();
            // 遍历角色ID列表，为每个角色ID创建用户角色关系并插入数据库
            roleIdList.forEach(roleId -> {
                // 创建一个新的用户角色关系对象
                SysUserRole sysUserRole = new SysUserRole();
                // 设置用户ID
                sysUserRole.setUserId(userId);
                // 设置角色ID
                sysUserRole.setRoleId(roleId);
                // 注意：不建议在循环中操作数据库，大家尽量需要避免在循环中操作数据库
                // sysUserRoleMapper.insert(sysUserRole);

                // 将创建的用户角色关系添加到列表中
                sysUserRoleList.add(sysUserRole);
            });
            // 批量保存用户角色关系
            sysUserRoleService.saveBatch(sysUserRoleList);
        }
        //修改管理员信息
        //获取新密码（如果有值：说明管理员修改了密码，如果没有值：说明原密码不变）
        // 获取用户的新密码
        String newPassword = sysUser.getPassword();
        // 检查新密码是否非空
        if (StringUtils.hasText(newPassword)) {
            // 对新密码进行加密处理
            sysUser.setPassword(passwordEncoder.encode(newPassword));
        }
        // 更新数据库中的用户密码
        return sysUserMapper.updateById(sysUser);
    }

    /**
     * 单个或批量删除管理员
     * 1.批量/单个删除管理员与角色的关系记录
     * 2.批量/单个删除管理员
     *
     * @param userIds
     * @return
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public Boolean removeSysUserListByUserIds(List<Long> userIds) {
        // 1. 批量/单个删除管理员与角色的关系记录
        sysUserRoleMapper.delete(new LambdaQueryWrapper<SysUserRole>()
                .in(SysUserRole::getUserId, userIds));
        // 2. 批量/单个删除管理员
        return sysUserMapper.deleteBatchIds(userIds) == userIds.size();
    }

}
